package com.xukai.springboot.util.config;

/**
 * 作用：
 * 1、保存一个线程安全的DatabaseType容器
 */
public class DatabaseContextHolder {
    /**
     * 默认数据源
     */
    public static final String DEFAULT_DS = "manger";

    private static final ThreadLocal<String> contextHolder = new ThreadLocal<>();

    // 设置数据源名
    public static void setDB(String dbType) {
        if (dbType.equals(getDB())) {
            System.out.println("数据源相同不需要进行数据源切换");
            return;
        }
        System.out.println("切换到{" + dbType + "}数据源");
        contextHolder.set(dbType);
    }

    // 获取数据源名
    public static String getDB() {
        return (contextHolder.get());
    }

    // 清除数据源名
    public static void clearDB() {
        contextHolder.remove();
    }


}